import requests
import json

def get_tag_page_info(tag: str = "flowchart"):
    """
    获取 tag 对应的 mermaid 页面信息
    args: 前端用户选定的 tag
        图表类型对应名称:  tag -> mermaid type name
        - 流程图 flowchart
        - 甘特图 gantt
        - 序列图 sequenceDiagram
        - 类图 classDiagram
        - 状态图 stateDiagram
        - 饼图 pie
        - 实体关系图  entityRelationshipDiagram
    return: mermaid 中午官网对应类型的介绍文档内容
    """
    url = "https://scrape.serper.dev"
    mermaid_url = f"https://mermaid.nodejs.cn/syntax/{tag}.html"
    # mermaid 地址示例：https://mermaid.nodejs.cn/syntax/flowchart.html, 其中 flowchart 就是可替换的 tag
    payload = json.dumps({
    "url": mermaid_url,
    "includeMarkdown": True # 是否返回 Markdown 格式
    })
    headers = {
    'X-API-KEY': '0b33d1ee4053a32b93f6d029f66aa149599d7c91', # Google Serper API Key, 目前是 weihua 的
    'Content-Type': 'application/json'
    }

    response = requests.request("POST", url, headers=headers, data=payload)
    return response.text

if __name__ == "__main__":
    print(get_tag_page_info())
